import Tkinter as tk
from random import randint
import math
from Tkinter import *

COLORS  =['snow', 'ghost white', 'white smoke', 'gainsboro', 'floral white', 'old lace',
    'linen', 'antique white', 'papaya whip', 'blanched almond', 'bisque', 'peach puff',
    'navajo white', 'lemon chiffon', 'mint cream', 'azure', 'alice blue', 'lavender',
    'lavender blush', 'misty rose', 'dark slate gray', 'dim gray', 'slate gray',
    'light slate gray', 'gray', 'light grey', 'midnight blue', 'navy', 'cornflower blue', 'dark slate blue',
    'slate blue', 'medium slate blue', 'light slate blue', 'medium blue', 'royal blue',  'blue',
    'dodger blue', 'deep sky blue', 'sky blue', 'light sky blue', 'steel blue', 'light steel blue',
    'light blue', 'powder blue', 'pale turquoise', 'dark turquoise', 'medium turquoise', 'turquoise',
    'cyan', 'light cyan', 'cadet blue', 'medium aquamarine', 'aquamarine', 'dark green', 'dark olive green',
    'dark sea green', 'sea green', 'medium sea green', 'light sea green', 'pale green', 'spring green',
    'lawn green', 'medium spring green', 'green yellow', 'lime green', 'yellow green',
    'forest green', 'olive drab', 'dark khaki', 'khaki', 'pale goldenrod', 'light goldenrod yellow',
    'light yellow', 'yellow', 'gold', 'light goldenrod', 'goldenrod', 'dark goldenrod', 'rosy brown',
    'indian red', 'saddle brown', 'sandy brown',
    'dark salmon', 'salmon', 'light salmon', 'orange', 'dark orange',
    'coral', 'light coral', 'tomato', 'orange red', 'red', 'hot pink', 'deep pink', 'pink', 'light pink',
    'pale violet red', 'maroon', 'medium violet red', 'violet red',
    'medium orchid', 'dark orchid', 'dark violet', 'blue violet', 'purple', 'medium purple',
    'thistle', 'snow2', 'snow3',
    'snow4', 'seashell2', 'seashell3', 'seashell4', 'AntiqueWhite1', 'AntiqueWhite2',
    'AntiqueWhite3', 'AntiqueWhite4', 'bisque2', 'bisque3', 'bisque4', 'PeachPuff2',
    'PeachPuff3', 'PeachPuff4', 'NavajoWhite2', 'NavajoWhite3', 'NavajoWhite4',
    'LemonChiffon2', 'LemonChiffon3', 'LemonChiffon4', 'cornsilk2', 'cornsilk3',
    'cornsilk4', 'ivory2', 'ivory3', 'ivory4', 'honeydew2', 'honeydew3', 'honeydew4',
    'LavenderBlush2', 'LavenderBlush3', 'LavenderBlush4', 'MistyRose2', 'MistyRose3',
    'MistyRose4', 'azure2', 'azure3', 'azure4', 'SlateBlue1', 'SlateBlue2', 'SlateBlue3',
    'SlateBlue4', 'RoyalBlue1', 'RoyalBlue2', 'RoyalBlue3', 'RoyalBlue4', 'blue2', 'blue4',
    'DodgerBlue2', 'DodgerBlue3', 'DodgerBlue4', 'SteelBlue1', 'SteelBlue2',
    'SteelBlue3', 'SteelBlue4', 'DeepSkyBlue2', 'DeepSkyBlue3', 'DeepSkyBlue4',
    'SkyBlue1', 'SkyBlue2', 'SkyBlue3', 'SkyBlue4', 'LightSkyBlue1', 'LightSkyBlue2',
    'LightSkyBlue3', 'LightSkyBlue4', 'SlateGray1', 'SlateGray2', 'SlateGray3',
    'SlateGray4', 'LightSteelBlue1', 'LightSteelBlue2', 'LightSteelBlue3',
    'LightSteelBlue4', 'LightBlue1', 'LightBlue2', 'LightBlue3', 'LightBlue4',
    'LightCyan2', 'LightCyan3', 'LightCyan4', 'PaleTurquoise1', 'PaleTurquoise2',
    'PaleTurquoise3', 'PaleTurquoise4', 'CadetBlue1', 'CadetBlue2', 'CadetBlue3',
    'CadetBlue4', 'turquoise1', 'turquoise2', 'turquoise3', 'turquoise4', 'cyan2', 'cyan3',
    'cyan4', 'DarkSlateGray1', 'DarkSlateGray2', 'DarkSlateGray3', 'DarkSlateGray4',
    'aquamarine2', 'aquamarine4', 'DarkSeaGreen1', 'DarkSeaGreen2', 'DarkSeaGreen3',
    'DarkSeaGreen4', 'SeaGreen1', 'SeaGreen2', 'SeaGreen3', 'PaleGreen1', 'PaleGreen2',
    'PaleGreen3', 'PaleGreen4', 'SpringGreen2', 'SpringGreen3', 'SpringGreen4',
    'green2', 'green3', 'green4', 'chartreuse2', 'chartreuse3', 'chartreuse4',
    'OliveDrab1', 'OliveDrab2', 'OliveDrab4', 'DarkOliveGreen1', 'DarkOliveGreen2',
    'DarkOliveGreen3', 'DarkOliveGreen4', 'khaki1', 'khaki2', 'khaki3', 'khaki4',
    'LightGoldenrod1', 'LightGoldenrod2', 'LightGoldenrod3', 'LightGoldenrod4',
    'LightYellow2', 'LightYellow3', 'LightYellow4', 'yellow2', 'yellow3', 'yellow4',
    'gold2', 'gold3', 'gold4', 'goldenrod1', 'goldenrod2', 'goldenrod3', 'goldenrod4',
    'DarkGoldenrod1', 'DarkGoldenrod2', 'DarkGoldenrod3', 'DarkGoldenrod4',
    'RosyBrown1', 'RosyBrown2', 'RosyBrown3', 'RosyBrown4', 'IndianRed1', 'IndianRed2',
    'IndianRed3', 'IndianRed4', 'sienna1', 'sienna2', 'sienna3', 'sienna4', 'burlywood1',
    'burlywood2', 'burlywood3', 'burlywood4', 'wheat1', 'wheat2', 'wheat3', 'wheat4', 'tan1',
    'tan2', 'tan4', 'chocolate1', 'chocolate2', 'chocolate3', 'firebrick1', 'firebrick2',
    'firebrick3', 'firebrick4', 'brown1', 'brown2', 'brown3', 'brown4', 'salmon1', 'salmon2',
    'salmon3', 'salmon4', 'LightSalmon2', 'LightSalmon3', 'LightSalmon4', 'orange2',
    'orange3', 'orange4', 'DarkOrange1', 'DarkOrange2', 'DarkOrange3', 'DarkOrange4',
    'coral1', 'coral2', 'coral3', 'coral4', 'tomato2', 'tomato3', 'tomato4', 'OrangeRed2',
    'OrangeRed3', 'OrangeRed4', 'red2', 'red3', 'red4', 'DeepPink2', 'DeepPink3', 'DeepPink4',
    'HotPink1', 'HotPink2', 'HotPink3', 'HotPink4', 'pink1', 'pink2', 'pink3', 'pink4',
    'LightPink1', 'LightPink2', 'LightPink3', 'LightPink4', 'PaleVioletRed1',
    'PaleVioletRed2', 'PaleVioletRed3', 'PaleVioletRed4', 'maroon1', 'maroon2',
    'maroon3', 'maroon4', 'VioletRed1', 'VioletRed2', 'VioletRed3', 'VioletRed4',
    'magenta2', 'magenta3', 'magenta4', 'orchid1', 'orchid2', 'orchid3', 'orchid4', 'plum1',
    'plum2', 'plum3', 'plum4', 'MediumOrchid1', 'MediumOrchid2', 'MediumOrchid3',
    'MediumOrchid4', 'DarkOrchid1', 'DarkOrchid2', 'DarkOrchid3', 'DarkOrchid4',
    'purple1', 'purple2', 'purple3', 'purple4', 'MediumPurple1', 'MediumPurple2',
    'MediumPurple3', 'MediumPurple4', 'thistle1', 'thistle2', 'thistle3', 'thistle4',
    'gray1', 'gray2', 'gray3', 'gray4', 'gray5', 'gray6', 'gray7', 'gray8', 'gray9', 'gray10',
    'gray11', 'gray12', 'gray13', 'gray14', 'gray15', 'gray16', 'gray17', 'gray18', 'gray19',
    'gray20', 'gray21', 'gray22', 'gray23', 'gray24', 'gray25', 'gray26', 'gray27', 'gray28',
    'gray29', 'gray30', 'gray31', 'gray32', 'gray33', 'gray34', 'gray35', 'gray36', 'gray37',
    'gray38', 'gray39', 'gray40', 'gray42', 'gray43', 'gray44', 'gray45', 'gray46', 'gray47',
    'gray48', 'gray49', 'gray50', 'gray51', 'gray52', 'gray53', 'gray54', 'gray55', 'gray56',
    'gray57', 'gray58', 'gray59', 'gray60', 'gray61', 'gray62', 'gray63', 'gray64', 'gray65',
    'gray66', 'gray67', 'gray68', 'gray69', 'gray70', 'gray71', 'gray72', 'gray73', 'gray74',
    'gray75', 'gray76', 'gray77', 'gray78', 'gray79', 'gray80', 'gray81', 'gray82', 'gray83',
    'gray84', 'gray85', 'gray86', 'gray87', 'gray88', 'gray89', 'gray90', 'gray91', 'gray92',
    'gray93', 'gray94', 'gray95', 'gray97', 'gray98', 'gray99']

root = tk.Tk()

class MainClass: 
    def __init__(self):
        self.window_width = 800   
        self.window_height = 600
        self.bodies = []
        self.force = [0.0,0.0,0.0]
        paddle = Body()
        paddle.radius = 20
        paddle.type = "paddle"        
        paddle.position[0]=420
        paddle.position[1]=540
        paddle.color = "black"
        self.addBody(paddle)
        cue_ball = Body()
        cue_ball.type = "circle"
        cue_ball.radius = 7
        cue_ball.position[0] = paddle.position[0]-30
        cue_ball.position[1] = paddle.position[1]-14
        cue_ball.color = "yellow"
        self.addBody(cue_ball)   
        for row in range(14):
            for col in range(14):
                body = Body()
                body.position = [49*row+98,20*col+68]
                body.radius = 20
                self.bodies.append(body)

    def addBody(self,body):
        body.id = len(self.bodies)+1
        self.bodies.append(body)    
    
    def draw(self):
        self.w.delete("all")
        self.w.create_rectangle(0,0,self.window_width,self.window_height,fill="honeydew2")
        self.w.create_line(self.window_width/2.0-350,self.window_height/2.0-250,self.window_width/2.0+350,self.window_height/2.0-250)
        self.w.create_line(self.window_width/2.0+350,self.window_height/2.0-250,self.window_width/2.0+350,self.window_height/2.0+250)
        self.w.create_line(self.window_width/2.0+350,self.window_height/2.0+250,self.window_width/2.0-350,self.window_height/2.0+250)
        self.w.create_line(self.window_width/2.0-350,self.window_height/2.0+250,self.window_width/2.0-350,self.window_height/2.0-250)
        for body in self.bodies:
            if body.type == "rectangle":
                x = body.position[0]
                y = body.position[1]
                r = body.radius
                self.w.create_rectangle(x-r*2,y-r/2,x+(r/3),y+(r/3),fill=body.color)
            elif body.type == "circle":
                x = body.position[0]
                y = body.position[1]
                r = body.radius
                self.w.create_oval(x-r,y-r,x+r,y+r,fill=body.color)
            else:
                x = body.position[0] 
                y = body.position[1]
                r = body.radius
                self.w.create_rectangle(x-r*3.5,y-r/3.5,x+(r/3),y+(r/3),fill=body.color)
                
    def secondary_function(self):
        print "This is another function."
        
    def left_key(self,event):
        x= self.bodies[0].position[0]
        a= self.bodies[1].position[0]
        if x <= 120:
            x +=0
            a +=0
        else:
            self.bodies[0].position[0] -=10
            self.bodies[1].position[0] -=10
            self.draw()
    def right_key(self,event):
        x= self.bodies[0].position[0]
        a= self.bodies[1].position[0]
        if x >=742:
            x += 0
            a += 0
        elif x == 740:
            x += 2
            a += 2
        else:
            self.bodies[0].position[0]+=10
            self.bodies[1].position[0]+=10
            self.draw()
    
    def applyForce(self,direction,magnitude):
        self.force[0] += magnitude * math.cos(direction)      
        self.force[1] += magnitude * math.sin(direction)    
    
    def space_bar(self,event):
        for bodies in self.bodies:
            self.bodies[0].applyForce(math.pi/2, 1000)
    

#    def step(self):
 #       self.current_step += 1
  #      self.time += self.h
        
        # Reset body forces
   #     for body in self.bodies:
    #        for i in range(len(body.acceleration)):
     #           body.force[i] = body.mass * body.acceleration[i]
        
        # Calculate forces, accelerations, and determine velocities
      #  for body1 in self.bodies:
       #     for body2 in self.bodies:
        #        if body1 != body2:
         #           todo = True
                    # TODO: For each body pair, apply a force if they are colliding 
          #          x = body1.position[0] - body2.position[0]        
           #         y = body1.position[1] - body2.position[1]                    
            #        distance = math.sqrt ((x)**2 + (y)**2)- body1.radius - body2.radius                   
             #       circle_angle = math.atan2(y,x)                    
              #      if distance <= 0:
               #         body1.applyForce(math.atan2(y,x),500)
                        
        
            
    def run(self):
        self.w = tk.Canvas(root, width=self.window_width, height=self.window_height)
        self.w.pack()
        self.w.after(0,self.draw)  
                        
        frame = Frame(root, width=100, height=100)
        root.bind('<Left>', self.left_key)
        root.bind('<Right>', self.right_key)
        root.bind('<space>', self.space_bar)
        frame.pack()
        root.mainloop
        tk.mainloop()
    
class Body():
    def __init__(self):
       self.type = "rectangle"
       self.position = [0.0,0.0,0.0]           
       self.dynamic = True                     
       self.radius = 100
       self.color = COLORS[randint(0,len(COLORS))-1]
    
    def wall_force(self):
        for body in self.bodies:
            # Left wall
            if body.position[0] <= -800 + body.radius:
                body.applyForce(0, 1000)
            # Ceiling
            if body.position[1] >= 600 - body.radius:
                body.applyForce(1.5 * math.pi, 1000)
            #Right wall
            if body.position[0] >= 800 - body.radius:
                body.applyForce(math.pi, 1000)



program = MainClass()
program.run()
#paddle = Body()
#paddle.run()





